from sqlalchemy.orm import relationship
from .user import User
from .knowledge_base import KnowledgeBase
from .mcp_service import MCPService
from .conversation import Conversation

# 更新关系
User.knowledge_bases = relationship("KnowledgeBase", back_populates="user")
User.conversations = relationship("Conversation", back_populates="user")

__all__ = ["User", "KnowledgeBase", "MCPService", "Conversation"]